#ifndef __LABELDISTRIBUTIONTEXTWRITER_H__
#define __LABELDISTRIBUTIONTEXTWRITER_H__

#include "LabelDistributionWriter.h"

#include <fstream>

class LabelDistributionTextWriter
	: public LabelDistributionWriter
{
	std::ofstream m_fOutput;

public:
	LabelDistributionTextWriter(std::string &sOutputFileName);
	virtual ~LabelDistributionTextWriter();

	virtual int Write(Node *key, ListLabel *val);
};

#endif